# 使用 CUDA 11.6 的基础镜像
FROM nvidia/cuda:11.6.2-runtime-ubuntu20.04

# 安装常用工具和依赖
# RUN apt-get update && apt-get install -y \
#     git \
#     wget \
#     && rm -rf /var/lib/apt/lists/*

# 安装 Miniconda（或者 Anaconda，根据您的选择）
RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O /tmp/miniconda.sh && \
    bash /tmp/miniconda.sh -b -p /opt/conda && \
    rm /tmp/miniconda.sh
ENV PATH="/opt/conda/bin:${PATH}"

# 创建并激活虚拟环境
RUN conda create -n myenv python=3.8
RUN echo "source activate myenv" > ~/.bashrc
ENV PATH /opt/conda/envs/myenv/bin:$PATH
RUN /bin/bash -c "source activate myenv"

# 安装 PyTorch 和 torchvision
RUN conda install -y pytorch==1.10.0 torchvision==0.11.1 torchaudio==0.10.0 cudatoolkit=11.6 -c pytorch

# 设置工作目录
WORKDIR /app

# 在这里添加您的应用程序代码和其他必要的步骤

# 指定启动命令
CMD ["bash"]
